Handle nested calls and function oneliners when chaining