Fix generator test assertions