Fixed tests to reflect new migrations.