diff --git a/srcpkgs/ssh-audit/template b/srcpkgs/ssh-audit/template
index aab2e5e46e7..bfb8ec4466b 100644
--- a/srcpkgs/ssh-audit/template
+++ b/srcpkgs/ssh-audit/template
@@ -1,18 +1,24 @@
 # Template file for 'ssh-audit'
 pkgname=ssh-audit
-version=2.3.0
-revision=2
+version=2.3.1
+revision=1
+build_style="python3-module"
+hostmakedepends="python3-setuptools"
 depends="python3"
+checkdepends="python3-pytest"
 short_desc="SSH server & client auditing"
 maintainer="Andrew Benson <abenson+void@gmail.com>"
 license="MIT"
 homepage="https://github.com/jtesta/ssh-audit"
 distfiles="https://github.com/jtesta/ssh-audit/archive/v${version}.tar.gz"
-checksum=776547591e7b69a2a8dcd1eaaac5d38321f2cb4a5de5f8e5a3e135b33236e812
+checksum=3d81c2620f53e7e0fe359e94524f28b58d53aa4e77292e2c9c555e174993d314
 python_version=3
 
-do_install() {
-	vbin ssh-audit.py ssh-audit
+do_check() {
+	PYTHONPATH=build/lib python3 -m pytest
+}
+
+post_install() {
 	vlicense LICENSE
 	vman ssh-audit.1
 }