Fixed a couple of migration issues, made migration test actually asser something