Fix tests migrations run
Use new base and public structure