Fix for last migration error