diff --git a/src/connectedComponents/ConnectedLayoutButton.js b/src/connectedComponents/ConnectedLayoutButton.js
index 23f44e70b..4842f5952 100644
--- a/src/connectedComponents/ConnectedLayoutButton.js
+++ b/src/connectedComponents/ConnectedLayoutButton.js
@@ -1,6 +1,6 @@
-import { connect } from 'react-redux';
 import { LayoutButton } from 'react-viewerbase';
 import OHIF from 'ohif-core';
+import { connect } from 'react-redux';
 
 const { setLayout } = OHIF.redux.actions;
 
@@ -22,7 +22,6 @@ const mapDispatchToProps = dispatch => {
         viewports.push({
           height: `${100 / rows}%`,
           width: `${100 / columns}%`,
-          plugin: 'cornerstone', // Temporary because right now switching back from VTK breaks things
         });
       }
       const layout = {