Fix tests
Fix test migrations