Fix broken testbenches