updated bug with graph