Fix a couple of function calls