test(generator): working finally... the generator works differently now, had to update tests