diff --git a/roles/bitbucket_dataset_restore/tasks/main.yml b/roles/bitbucket_dataset_restore/tasks/main.yml
index c0fe43c..cad7f13 100644
--- a/roles/bitbucket_dataset_restore/tasks/main.yml
+++ b/roles/bitbucket_dataset_restore/tasks/main.yml
@@ -33,7 +33,7 @@
   until: output.status == 200
   retries: 6
   delay: 15
-  failed_when: output.json.state != 'INITIALISING'
+  failed_when: output.state != 'INITIALISING'
   #failed_when: "'INITIALISING' not in output.content"
 
 - name: get status of import