Fix messed up argument types