Adapt for new graph initialisation method