Added schema check to update