diff --git a/pkg/project/affected_test.go b/pkg/project/affected_test.go
index 0a8beec..938a4db 100644
--- a/pkg/project/affected_test.go
+++ b/pkg/project/affected_test.go
@@ -129,12 +129,12 @@ func TestIsAffected_DiffHash_ReturnTrue(t *testing.T) {
 }
 
 func TestStringToTarget_Build(t *testing.T) {
-	result := StringToTarget("build")
+	result := StringToTarget(TargetBuild.String())
 	assert.Equal(t, TargetBuild, result)
 }
 
 func TestStringToTarget_BuildImage(t *testing.T) {
-	result := StringToTarget("build-image")
+	result := StringToTarget(TargetBuildImage.String())
 	assert.Equal(t, TargetBuildImage, result)
 }