fix list comp bug