Fix return value