Forgot to clean up code