diff --git a/src/main/kotlin/Main.kt b/src/main/kotlin/Main.kt
index a1652d5..ce3847b 100644
--- a/src/main/kotlin/Main.kt
+++ b/src/main/kotlin/Main.kt
@@ -1,6 +1,8 @@
 import org.flywaydb.core.Flyway
 import org.hibernate.cfg.Configuration
 import javax.persistence.*
+import javax.sql.DataSource
+
 
 @Entity
 class Entry(