Fix small things

Good remark! Changed migration.