Forgot a couple changes