Fix query mutation update checks