Kevin Hu
fix: keyerror issue (#3512)
cf6b668