Fix return error