Forgot a null check