diff --git a/flask_reverse_proxy_fix/middleware/__init__.py b/flask_reverse_proxy_fix/middleware/__init__.py index 5ea641b..e914a00 100644 --- a/flask_reverse_proxy_fix/middleware/__init__.py +++ b/flask_reverse_proxy_fix/middleware/__init__.py @@ -1,6 +1,9 @@ from flask import Flask as App # noinspection PyPackageRequirements -from werkzeug.contrib.fixers import ProxyFix +try: + from werkzeug.contrib.fixers import ProxyFix +except ModuleNotFoundError: + from werkzeug.middleware.proxy_fix import ProxyFix class ReverseProxyPrefixFix(object):