diff --git a/de.prob.units/src/de/prob/units/sc/MachineAttributeProcessor.java b/de.prob.units/src/de/prob/units/sc/MachineAttributeProcessor.java
index c6d81b406b69abb98f6ac041a8b2ee7f34e00d8e..4c7ddd38f4a195674c5e270338b5031f1fd3f4a0 100644
--- a/de.prob.units/src/de/prob/units/sc/MachineAttributeProcessor.java
+++ b/de.prob.units/src/de/prob/units/sc/MachineAttributeProcessor.java
@@ -3,9 +3,9 @@ package de.prob.units.sc;
 import org.eclipse.core.runtime.CoreException;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eventb.core.IMachineRoot;
+import org.eventb.core.ISCMachineRoot;
 import org.eventb.core.ISCVariable;
 import org.eventb.core.IVariable;
-import org.eventb.core.basis.SCMachineRoot;
 import org.eventb.core.sc.SCCore;
 import org.eventb.core.sc.SCProcessorModule;
 import org.eventb.core.sc.state.ISCStateRepository;
@@ -26,13 +26,13 @@ public class MachineAttributeProcessor extends SCProcessorModule {
 			ISCStateRepository repository, IProgressMonitor monitor)
 			throws CoreException {
 		assert (element instanceof IRodinFile);
-		assert (target instanceof SCMachineRoot);
+		assert (target instanceof ISCMachineRoot);
 
 		// get all variables and copy over the attributes
 		IRodinFile machineFile = (IRodinFile) element;
 		IMachineRoot machineRoot = (IMachineRoot) machineFile.getRoot();
 
-		SCMachineRoot scMachineRoot = (SCMachineRoot) target;
+		ISCMachineRoot scMachineRoot = (ISCMachineRoot) target;
 
 		IVariable[] variables = machineRoot.getVariables();