Another optimization in the same function