Fix for wrong migration