fixed list comp error

had one too many for loops embedded.