From 1d552c27a53697f54b96d788978ee707f53e2307 Mon Sep 17 00:00:00 2001
From: Alan Garny <agarny@hellix.com>
Date: Wed, 5 Feb 2020 13:32:32 +1300
Subject: [PATCH] Some minor cleaning up.

---
 .../src/simulationexperimentviewsimulationwidget.cpp        | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/plugins/simulation/SimulationExperimentView/src/simulationexperimentviewsimulationwidget.cpp b/src/plugins/simulation/SimulationExperimentView/src/simulationexperimentviewsimulationwidget.cpp
index 0b0640be7f..e95eaf0130 100644
--- a/src/plugins/simulation/SimulationExperimentView/src/simulationexperimentviewsimulationwidget.cpp
+++ b/src/plugins/simulation/SimulationExperimentView/src/simulationexperimentviewsimulationwidget.cpp
@@ -1268,11 +1268,9 @@ bool SimulationExperimentViewSimulationWidget::save(const QString &pFileName)
             if (   (parameter->type() == CellMLSupport::CellmlFileRuntimeParameter::Type::State)
                 || (parameter->type() == CellMLSupport::CellmlFileRuntimeParameter::Type::Constant)) {
                 ObjRef<iface::cellml_api::CellMLComponent> component = components->getComponent(parameter->componentHierarchy().last().toStdWString());
-                ObjRef<iface::cellml_api::CellMLVariableSet>  variables = component->variables();
-                ObjRef<iface::cellml_api::CellMLVariable> variable = variables->getVariable(property->name().toStdWString());
-                ObjRef<iface::cellml_api::CellMLVariable> sourceVariable = variable->sourceVariable();
+                ObjRef<iface::cellml_api::CellMLVariable> variable = component->variables()->getVariable(property->name().toStdWString());
 
-                if (variable == sourceVariable) {
+                if (variable == variable->sourceVariable()) {
                     variable->initialValue(property->value().toStdWString());
                 } else {
                     importedParameters += "\n - "+QString::fromStdWString(component->name())+" | "+QString::fromStdWString(variable->name());