A little change in breakout_testbench.py