Fix return value
