Pass current generator to tests