modification to the test bench