Fix schema load issue