diff --git a/roles/bitbucket_dataset_restore/tasks/main.yml b/roles/bitbucket_dataset_restore/tasks/main.yml
index 5abc720..471a9f2 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 is defined and output.json.state != 'INITIALISING'
+  failed_when: output is defined and output.json is defined and output.json.state == 'INITIALISING'
   #failed_when: "'INITIALISING' not in output.content"
 
 - name: get status of import