diff --git a/devel/py-codegen/Makefile b/devel/py-codegen/Makefile
index 7e21ffdad045..7aae2067b4df 100644
--- a/devel/py-codegen/Makefile
+++ b/devel/py-codegen/Makefile
@@ -1,17 +1,20 @@
 PORTNAME=	codegen
 PORTVERSION=	1.0
 PORTREVISION=	1
 CATEGORIES=	devel python
 MASTER_SITES=	PYPI
 PKGNAMEPREFIX=	${PYTHON_PKGNAMEPREFIX}
 
 MAINTAINER=	antoine@FreeBSD.org
 COMMENT=	Extension to ast that allows ast to python code generation
 WWW=		https://github.com/andreif/codegen
 
 LICENSE=	BSD3CLAUSE
 
+BUILD_DEPENDS=	${PY_SETUPTOOLS} \
+		${PYTHON_PKGNAMEPREFIX}wheel>=0:devel/py-wheel@${PY_FLAVOR}
+
 USES=		python
-USE_PYTHON=	distutils autoplist
+USE_PYTHON=	pep517 autoplist
 
 .include <bsd.port.mk>