Fixing function definition

Accidentally switched those during the refactor