Update to work with fake random test method