Forgot a small piece of changes