diff --git a/setup.py b/setup.py
index 6a83585..5d9962f 100644
--- a/setup.py
+++ b/setup.py
@@ -13,7 +13,7 @@
 
 setup(
     name='itk-ndreg',
-    version='0.1.0',
+    version='0.1.1',
     author='Neurodata',
     author_email='vikramc@jhmi.edu',
     packages=['itk'],
@@ -44,6 +44,6 @@
     keywords='ITK InsightToolkit',
     url=r'https://itk.org/',
     install_requires=[
-        r'itk'
+        r'itk>=5.0.1'
     ]
     )
diff --git a/test/azure-pipelines.yml b/test/azure-pipelines.yml
index 1d12788..07de04c 100644
--- a/test/azure-pipelines.yml
+++ b/test/azure-pipelines.yml
@@ -1,6 +1,6 @@
 variables:
-  ITKGitTag: v5.0.0
-  ITKPythonGitTag: v5.0.0.post1
+  ITKGitTag: v5.0.1
+  ITKPythonGitTag: v5.0.1
   CMakeBuildType: Release
 
 trigger:
@@ -58,7 +58,9 @@ jobs:
     displayName: 'Install build dependencies'
 
   - script: |
-      git clone --depth 5 --branch $(ITKGitTag) https://github.com/InsightSoftwareConsortium/ITK.git
+      git clone https://github.com/InsightSoftwareConsortium/ITK.git
+      cd ITK
+      git checkout $(ITKGitTag)
     workingDirectory: $(Agent.BuildDirectory)
     displayName: 'Download ITK'