Forgot a tiny fix