diff --git a/src/barabadb/core/server.nim b/src/barabadb/core/server.nim
index cef772e..d29e1dc 100644
--- a/src/barabadb/core/server.nim
+++ b/src/barabadb/core/server.nim
@@ -2,6 +2,7 @@
 import std/asyncdispatch
 import std/asyncnet
 import std/strutils
+import std/tables
 import std/os
 import std/endians
 import config
diff --git a/src/barabadb/query/parser.nim b/src/barabadb/query/parser.nim
index 52c151d..69984a6 100644
--- a/src/barabadb/query/parser.nim
+++ b/src/barabadb/query/parser.nim
@@ -774,7 +774,7 @@ proc parseCreateMigration(p: var Parser): Node =
                 line: tok.line, col: tok.col)
 
 proc parseApplyMigration(p: var Parser): Node =
-  let tok = p.expect(tkIdent)  # APPLY
+  let tok = p.expect(tkApply)
   discard p.expect(tkMigration)
   let name = p.expect(tkIdent).value
   result = Node(kind: nkApplyMigration, amName: name, line: tok.line, col: tok.col)