diff --git a/dev-python/python-betterproto/python-betterproto-2.0.0_beta4.ebuild b/dev-python/python-betterproto/python-betterproto-2.0.0_beta4.ebuild
index c8419dce04..83d67433e8 100644
--- a/dev-python/python-betterproto/python-betterproto-2.0.0_beta4.ebuild
+++ b/dev-python/python-betterproto/python-betterproto-2.0.0_beta4.ebuild
@@ -42,6 +42,11 @@ distutils_enable_tests pytest
 distutils_enable_sphinx docs \
 	'>=dev-python/sphinx_rtd_theme-0.5.0'
 
+python_prepare() {
+	default
+	python_fix_shebang tests/generate.py src/betterproto/plugin/main.py
+}
+
 python_test() {
 	"${EPYTHON}" -m tests.generate -v
 	epytest
diff --git a/dev-python/python-betterproto/python-betterproto-9999.ebuild b/dev-python/python-betterproto/python-betterproto-9999.ebuild
index c8419dce04..83d67433e8 100644
--- a/dev-python/python-betterproto/python-betterproto-9999.ebuild
+++ b/dev-python/python-betterproto/python-betterproto-9999.ebuild
@@ -42,6 +42,11 @@ distutils_enable_tests pytest
 distutils_enable_sphinx docs \
 	'>=dev-python/sphinx_rtd_theme-0.5.0'
 
+python_prepare() {
+	default
+	python_fix_shebang tests/generate.py src/betterproto/plugin/main.py
+}
+
 python_test() {
 	"${EPYTHON}" -m tests.generate -v
 	epytest