Missed 2 line change to logicalStep class.