Fix issue in helpers.py