Allow wrapper to accept sub_name only instead of sub