Fix kernel argument types