diff --git a/roles/bitbucket_dataset_restore/tasks/main.yml b/roles/bitbucket_dataset_restore/tasks/main.yml
index 7b31097..ca0869b 100644
--- a/roles/bitbucket_dataset_restore/tasks/main.yml
+++ b/roles/bitbucket_dataset_restore/tasks/main.yml
@@ -33,11 +33,14 @@
   until: response.status == 200
   retries: 6
   delay: 15
-  failed_when: response['response']['json']['state'] != 'INITIALISING'
+  failed_when: response['json']['state'] != 'INITIALISING'
 
 - name: get status of import
   debug: var=response
 
 - name: get job id
-  debug: var=response['response']['json']['id']
+  debug: var=response['json']['state']
+
+- name: get job id
+  debug: var=response['json']['id']