Fix for documentation (check_units was wiping out args for functions)