diff --git a/pkgs/development/python-modules/mdformat-beautysh/default.nix b/pkgs/development/python-modules/mdformat-beautysh/default.nix
index 522494f783fe..547d289f6f95 100644
--- a/pkgs/development/python-modules/mdformat-beautysh/default.nix
+++ b/pkgs/development/python-modules/mdformat-beautysh/default.nix
@@ -6,7 +6,9 @@
 , mdformat-gfm
 , mdit-py-plugins
 , poetry-core
+, pytestCheckHook
 , pythonOlder
+, wheel
 }:
 
 buildPythonPackage rec {
@@ -25,6 +27,7 @@ buildPythonPackage rec {
 
   nativeBuildInputs = [
     poetry-core
+    wheel
   ];
 
   buildInputs = [
@@ -37,6 +40,14 @@ buildPythonPackage rec {
     beautysh
   ];
 
+  nativeCheckInputs = [
+    pytestCheckHook
+  ];
+
+  pythonImportsCheck = [
+    "mdformat_beautysh"
+  ];
+
   meta = with lib; {
     description = "Mdformat plugin to beautify Bash scripts";
     homepage = "https://github.com/hukkin/mdformat-beautysh";