Minor patch to graph (forgot to fix from last merge) (#27)

Minor patch to graph (forgot to fix from last merge) (#27) #476