fix graph behaviour
Merge pull request #59 from IGI-111/graph

Fix graph behaviour