Forgot to update a function