More tests for test gen