Add key error protection