Another small fix on graphs