migration correction - just 2 models