Updates testbench for wb_s module