Forgot to change the function