diff --git a/lib/cli/writer.js b/lib/cli/writer.js
index 8125d9a7..9da73224 100644
--- a/lib/cli/writer.js
+++ b/lib/cli/writer.js
@@ -23,7 +23,7 @@ const os = require('os');
 const unmount = require('./unmount');
 const imageStream = require('../image-stream');
 const errors = require('../shared/errors');
-const constraints = require('../shared/constraints');
+const constraints = require('../shared/drive-constraints');
 
 /**
  * @summary Write an image to a disk drive