Fix for chisel 2.0

as per title