Fixed dict.keys() index error