diff --git a/hoptimator-k8s/src/main/java/com/linkedin/hoptimator/k8s/K8sEngine.java b/hoptimator-k8s/src/main/java/com/linkedin/hoptimator/k8s/K8sEngine.java
index 6cbdeea..c0dd7ec 100644
--- a/hoptimator-k8s/src/main/java/com/linkedin/hoptimator/k8s/K8sEngine.java
+++ b/hoptimator-k8s/src/main/java/com/linkedin/hoptimator/k8s/K8sEngine.java
@@ -35,6 +35,6 @@ public DataSource dataSource() {
 
   @Override
   public SqlDialect dialect() {
-    return SqlDialect.FLINK;  // TODO fix hardcoded dialect
+    return dialect;
   }
 }
diff --git a/hoptimator-k8s/src/main/java/com/linkedin/hoptimator/k8s/K8sEngineTable.java b/hoptimator-k8s/src/main/java/com/linkedin/hoptimator/k8s/K8sEngineTable.java
index 6026308..0b12d7f 100644
--- a/hoptimator-k8s/src/main/java/com/linkedin/hoptimator/k8s/K8sEngineTable.java
+++ b/hoptimator-k8s/src/main/java/com/linkedin/hoptimator/k8s/K8sEngineTable.java
@@ -5,7 +5,6 @@
 import java.util.Locale;
 import java.util.Optional;
 import java.util.stream.Collectors;
-import javax.sql.DataSource;
 
 import org.apache.calcite.adapter.jdbc.JdbcSchema;
 import org.apache.calcite.schema.Schema;