Add an extra check that was missing