Fix to get graph running.