Fix automigrate tests