Fixed error in migration