[~CLEANUP] correct function parameter types