diff --git a/src/app/service-graph/index.ts b/src/app/service-graph/index.ts
index 0bed24b..334f90a 100644
--- a/src/app/service-graph/index.ts
+++ b/src/app/service-graph/index.ts
@@ -23,6 +23,7 @@
 import {XosGraphConfig} from './services/graph.config';
 import {XosNodeRenderer} from './services/renderer/node.renderer';
 import {XosServiceGraphLegend} from './components/graph-legend/graph-legend.component';
+import {XosGraphStateMachine} from './services/graph-state-machine';
 
 export const xosServiceGraph = 'xosServiceGraph';
 
@@ -34,6 +35,7 @@
   .service('XosServiceGraphIcons', XosServiceGraphIcons)
   .service('XosNodePositioner', XosNodePositioner)
   .service('XosGraphConfig', XosGraphConfig)
+  .service('XosGraphStateMachine', XosGraphStateMachine)
   .service('XosNodeRenderer', XosNodeRenderer)
   .component('xosServiceGraph', XosServiceGraph)
   .component('xosServiceGraphLegend', XosServiceGraphLegend)