Forgot a little cleanup