Remove references to old graph module