Forgot to change return type