from soc.alu.input_stage import ALUInputStage
 from soc.alu.pipe_data import ALUPipeSpec
-from soc.alu.alu_input_record import CompALUOpSubset
+from soc.branch.br_input_record import CompBROpSubset
 from soc.decoder.power_enums import InternalOp
 import unittest
 
         m = Module()
         comb = m.d.comb
 
-        rec = CompALUOpSubset()
+        rec = CompBROpSubset()
         recwidth = 0
         # Setup random inputs for dut.op
         for p in rec.ports():
                  dut.i.b.eq(b),
                  a.eq(AnyConst(64)),
                  b.eq(AnyConst(64))]
-                      
 
         comb += dut.i.ctx.op.eq(rec)
 
-
         # Assert that op gets copied from the input to output
         for p in rec.ports():
             name = p.name
         with m.Else():
             comb += Assert(dut.o.b == b)
 
-
-
-
         return m
 
 class GTCombinerTestCase(FHDLTestCase):
 
 
 
 from soc.branch.pipeline import BranchBasePipe
-from soc.alu.alu_input_record import CompALUOpSubset
+from soc.branch.br_input_record import CompBROpSubset
 from soc.alu.pipe_data import ALUPipeSpec
 import random
 
         
 
     def test_ilang(self):
-        rec = CompALUOpSubset()
+        rec = CompBROpSubset()
 
         pspec = ALUPipeSpec(id_wid=2, op_wid=get_rec_width(rec))
         alu = BranchBasePipe(pspec)
 
         m.submodules.pdecode2 = pdecode2 = PowerDecode2(pdecode)
 
-        rec = CompALUOpSubset()
+        rec = CompBROpSubset()
 
         pspec = ALUPipeSpec(id_wid=2, op_wid=get_rec_width(rec))
         m.submodules.branch = branch = BranchBasePipe(pspec)