Forgot to move return outside brackets