diff --git a/azure-mgmt-compute/HISTORY.rst b/azure-mgmt-compute/HISTORY.rst
index 269ade956ebc..fc0f182a6ecb 100644
--- a/azure-mgmt-compute/HISTORY.rst
+++ b/azure-mgmt-compute/HISTORY.rst
@@ -3,6 +3,13 @@
 Release History
 ===============
 
+4.0.1 (2018-07-23)
+++++++++++++++++++
+
+**Bugfix**
+
+- Fix incorrect import from azure.mgmt.compute.models
+
 4.0.0 (2018-07-20)
 ++++++++++++++++++
 
diff --git a/azure-mgmt-compute/azure/mgmt/compute/models.py b/azure-mgmt-compute/azure/mgmt/compute/models.py
index 3f996e628912..eb8b45d7db77 100644
--- a/azure-mgmt-compute/azure/mgmt/compute/models.py
+++ b/azure-mgmt-compute/azure/mgmt/compute/models.py
@@ -7,8 +7,8 @@
 import warnings
 
 from .v2017_09_01.models import *
-from .v2017_12_01.models import *
-from .v2018_04_01.models import * # Note that this line is overriding some models of 2017-12-01. See link below for details.
+from .v2018_04_01.models import *
+from .v2018_06_01.models import * # Note that this line is overriding some models of 2018-04-01. See link below for details.
 
 warnings.warn("Import models from this file is deprecated. See https://aka.ms/pysdkmodels",
               DeprecationWarning)
diff --git a/azure-mgmt-compute/azure/mgmt/compute/version.py b/azure-mgmt-compute/azure/mgmt/compute/version.py
index 5a310ec590fb..ffdd8a201ce0 100644
--- a/azure-mgmt-compute/azure/mgmt/compute/version.py
+++ b/azure-mgmt-compute/azure/mgmt/compute/version.py
@@ -5,4 +5,4 @@
 # license information.
 # --------------------------------------------------------------------------
 
-VERSION = "4.0.0"
+VERSION = "4.0.1"
diff --git a/azure-mgmt-compute/tests/test_mgmt_compute.py b/azure-mgmt-compute/tests/test_mgmt_compute.py
index 84a161f417a2..793eb07c8974 100644
--- a/azure-mgmt-compute/tests/test_mgmt_compute.py
+++ b/azure-mgmt-compute/tests/test_mgmt_compute.py
@@ -517,6 +517,30 @@ def test_vm_sizes(self):
         virtual_machine_sizes = list(self.compute_client.virtual_machine_sizes.list(location))
         self.assertGreater(len(virtual_machine_sizes), 0)
 
+    def test_run_command(self):
+        # FIXME, test unfinished
+        run_commands_models = self.compute_client.virtual_machines.models
+
+        run_command_parameters = run_commands_models.RunCommandInput(
+            command_id="RunShellScript",
+            script=[
+                'echo $1 $2'
+            ],
+            parameters=[
+                run_commands_models.RunCommandInputParameter(name="arg1", value="hello"),
+                run_commands_models.RunCommandInputParameter(name="arg2", value="world"),
+            ]
+        )
+
+        run_command_parameters = {
+            'command_id': 'RunShellScript',
+            'script': [
+                'echo $arg1'
+            ],
+            'parameters': [
+                {'name':"arg1", 'value':"hello world"}
+            ]
+        }
 
 #------------------------------------------------------------------------------
 if __name__ == '__main__':