Fix schema bug