Reimplement tests with Generative gem