Fix tests migration