Fix return arguments