Fixed the test bench