Fixing problem with migration generation.