Fix schema update problem