Closes #51: Fix migration