Fix Migrations tests