Make getTaskStep logic public