Adding a test for generation