fix bug in mmu/fsm.py