Fixed function that didn't get updated