diff --git a/pkgs/development/python-modules/python-vagrant/default.nix b/pkgs/development/python-modules/python-vagrant/default.nix
index 694df9918e19..9984931e3a98 100644
--- a/pkgs/development/python-modules/python-vagrant/default.nix
+++ b/pkgs/development/python-modules/python-vagrant/default.nix
@@ -3,6 +3,7 @@
 buildPythonPackage rec {
   version = "1.0.0";
   pname = "python-vagrant";
+  format = "pyproject";
 
   src = fetchPypi {
     inherit pname version;
@@ -12,6 +13,10 @@ buildPythonPackage rec {
   # The tests try to connect to qemu
   doCheck = false;
 
+  pythonImportsCheck = [
+    "vagrant"
+  ];
+
   meta = {
     description = "Python module that provides a thin wrapper around the vagrant command line executable";
     homepage = "https://github.com/todddeluca/python-vagrant";