Fixed problem with migrations