Fix return argument