fixes following main generator test