diff --git a/client-java/sql/src/main/java/org/evomaster/client/java/sql/internal/SqlHandler.java b/client-java/sql/src/main/java/org/evomaster/client/java/sql/internal/SqlHandler.java
index 446ee05bef..650cfc1d23 100644
--- a/client-java/sql/src/main/java/org/evomaster/client/java/sql/internal/SqlHandler.java
+++ b/client-java/sql/src/main/java/org/evomaster/client/java/sql/internal/SqlHandler.java
@@ -259,7 +259,7 @@ But SELECT could be complex with many JOINs... whereas DIP would be simple(r)?
             throw new RuntimeException(e);
         }
 
-
+        //TODO isn't this a bug, as we do not pass the schema???
         return HeuristicsCalculator.computeDistance(command, data);
     }
 
diff --git a/core/src/main/kotlin/org/evomaster/core/output/service/TestSuiteWriter.kt b/core/src/main/kotlin/org/evomaster/core/output/service/TestSuiteWriter.kt
index 0bc874b536..368102fa8a 100644
--- a/core/src/main/kotlin/org/evomaster/core/output/service/TestSuiteWriter.kt
+++ b/core/src/main/kotlin/org/evomaster/core/output/service/TestSuiteWriter.kt
@@ -363,7 +363,7 @@ class TestSuiteWriter {
 
 
             if(solution.hasAnySqlAction()) {
-                addImport("org.evomaster.client.java.controller.db.dsl.SqlDsl.sql", lines, true)
+                addImport("org.evomaster.client.java.sql.dsl.SqlDsl.sql", lines, true)
                 addImport("org.evomaster.client.java.controller.api.dto.database.operations.InsertionResultsDto", lines)
                 addImport(InsertionDto::class.qualifiedName!!, lines)
             }