fixed list comp bug