Fix broken test and replace redundant generator with a tuple
1 file changed