Forgot to remove a couple of lines