fix: key error for graph