Fixed migration

Wrong schema function used