Key error fix