Remove old reference to optimization.graph module