fix dict comprehension test