Missed the tests in test_generate.py