diff --git a/scst/src/dev_handlers/scst_vdisk.c b/scst/src/dev_handlers/scst_vdisk.c
index 418cf3eaa..d7e581329 100644
--- a/scst/src/dev_handlers/scst_vdisk.c
+++ b/scst/src/dev_handlers/scst_vdisk.c
@@ -2338,7 +2338,7 @@ static inline void blockio_check_finish(struct scst_blockio_work *blockio_work)
 	if (atomic_dec_and_test(&blockio_work->bios_inflight)) {
 		blockio_work->cmd->completed = 1;
 		blockio_work->cmd->scst_cmd_done(blockio_work->cmd,
-			SCST_CMD_STATE_DEFAULT, SCST_CONTEXT_DIRECT_ATOMIC);
+			SCST_CMD_STATE_DEFAULT, scst_estimate_context());
 		kmem_cache_free(blockio_work_cachep, blockio_work);
 	}
 	return;