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