Fixed function calls that were changed in the last commit