diff --git a/roles/bitbucket_dataset_restore/tasks/main.yml b/roles/bitbucket_dataset_restore/tasks/main.yml
index c60dd73..2b9fdd1 100644
--- a/roles/bitbucket_dataset_restore/tasks/main.yml
+++ b/roles/bitbucket_dataset_restore/tasks/main.yml
@@ -61,4 +61,4 @@
   file: 
     path: "{{ atl_product_home_shared }}/data/migration/import/lock.file"
     state: touch
-  when: import_status.state == 'COMPLETED'
\ No newline at end of file
+  when: import_status.json.state == 'COMPLETED'
\ No newline at end of file