add generator switch to tests