Fix output for functions with 2 returns.