Fixed output to match the Python model. The state after scheduling is ok.