Forgot to add new arguments to function call in last commit.