Fixed problem with migrations. Fixes #17