diff --git a/packages/entity/src/entity-renderer/view/index.jsx b/packages/entity/src/entity-renderer/view/index.jsx
index fdf7bae..a2f482c 100644
--- a/packages/entity/src/entity-renderer/view/index.jsx
+++ b/packages/entity/src/entity-renderer/view/index.jsx
@@ -3,7 +3,7 @@ import './index.scss';
 import {PropTypes, React} from '@latus/react';
 
 import useEntity from '../../hooks/use-entity';
-import Stage from '../stage';
+import EntityStage from '../stage';
 
 const View = ({json}) => {
   const entity = useEntity(json);
@@ -19,7 +19,7 @@ const View = ({json}) => {
     );
   }
   else {
-    output = <Stage entity={entity} />;
+    output = <EntityStage entity={entity} />;
   }
   return (
     <div className="entity-view">