Fix function return