diff --git a/textproc/py-sentencepiece/Makefile b/textproc/py-sentencepiece/Makefile
index 083154938273..03109ff10315 100644
--- a/textproc/py-sentencepiece/Makefile
+++ b/textproc/py-sentencepiece/Makefile
@@ -1,27 +1,28 @@
 PORTNAME=	sentencepiece
 DISTVERSIONPREFIX=	v
 DISTVERSION=	0.2.0
 CATEGORIES=	textproc # machine-learning
 PKGNAMEPREFIX=	${PYTHON_PKGNAMEPREFIX}
 
 MAINTAINER=	yuri@FreeBSD.org
 COMMENT=	Unsupervised text tokenizer for Neural Network-based text generation
 WWW=		https://github.com/google/sentencepiece
 
 LICENSE=	APACHE20
 LICENSE_FILE=	${WRKSRC}/../LICENSE
 
 BUILD_DEPENDS=	protobuf>0:devel/protobuf
 LIB_DEPENDS=	libsentencepiece.so:textproc/sentencepiece
 
 USES=		compiler:c++17-lang pkgconfig python
-USE_PYTHON=	distutils autoplist pytest # many tests fail, see https://github.com/google/sentencepiece/issues/977
+USE_PYTHON=	distutils autoplist
 
 USE_GITHUB=	yes
 GH_ACCOUNT=	google
 
 WRKSRC_SUBDIR=	python
 
-TEST_ENV=	${MAKE_ENV} PYTHONPATH=${STAGEDIR}${PYTHONPREFIX_SITELIBDIR}
+do-test:
+	@cd ${WRKSRC} && ${SETENV} ${TEST_ENV} ${PYTHON_CMD} ${PYSETUP} test
 
 .include <bsd.port.mk>