Modify testbench