Small fix for key not exist warning