Fix return spec