Forgot to remove a bit of code