Updated tb to match fsm.