Forgot to re-add this check