fixing test so that code will compile