Fix schema migration