Fix tests for python 2.6 (don't have dict comp.)