Updated the test bench for the new 1024-unit machine