Dostosowanie do nowej wersji generatora