diff --git a/pkg/tuning/preset-tuning_test.go b/pkg/tuning/preset-tuning_test.go
index 9d5606b80..b2d7d4143 100644
--- a/pkg/tuning/preset-tuning_test.go
+++ b/pkg/tuning/preset-tuning_test.go
@@ -276,7 +276,7 @@ func TestHandleImageDataSource(t *testing.T) {
 
 	for name, tc := range testcases {
 		t.Run(name, func(t *testing.T) {
-			initContainer, volume, volumeMount := handleImageDataSource(context.Background(), tc.workspaceObj)
+			initContainer, volume, volumeMount := handleImageDataSource(context.Background(), tc.workspaceObj.Tuning.Input.Image)
 
 			assert.Equal(t, tc.expectedInitContainerName, initContainer.Name)
 			assert.Equal(t, tc.workspaceObj.Tuning.Input.Image, initContainer.Image)