Fixed bug with graph loading