fixing test and rerecording mock