Fixed a un-pythonic list access. How did I even managed to put it that way!!!