Fix return calc