diff --git a/smi-conformance/test-gen/test_gen.go b/smi-conformance/test-gen/test_gen.go
index b8e5e90..f46c42d 100644
--- a/smi-conformance/test-gen/test_gen.go
+++ b/smi-conformance/test-gen/test_gen.go
@@ -62,6 +62,10 @@ func RunTest(meshConfig ServiceMesh, annotations, labels map[string]string) Resu
 			SMObj: meshConfig,
 		}
 
+		if len(args) != 0 {
+			options.TestDirs = args
+		}
+
 		testHandlers := make(map[string]map[string]test.CustomTest)
 		testHandlers["traffic-access"] = serviceMeshConfObj.TrafficAccessGetTests()
 		testHandlers["traffic-spec"] = serviceMeshConfObj.TrafficSpecGetTests()
@@ -75,9 +79,6 @@ func RunTest(meshConfig ServiceMesh, annotations, labels map[string]string) Resu
 				NamespaceAnnotations: annotations,
 				NamespaceLabels:      labels,
 			}
-			if len(args) != 0 {
-				options.TestDirs = args
-			}
 
 			// Runs the test using the inCluster kubeConfig (runs only when the code is running inside the pod)
 			harness.InCluster = true