Fixed the test bench. Hopefully ready to roll