ModuleNotFoundError while following the DVC course

I am following the Iterative Tools for Data Scientists & Analysts course step by step, so far everything was going smoothly, but now I am at task 3.5 and when I am trying to follow along the course and execute dvc repro I get:

ModuleNotFoundError: No module named 'src'
ERROR: failed to reproduce 'dvc.yaml': failed to run: python src/stages/data_load.py --config=params.yaml, exited with 1

I am having same branch and files as in the course so not sure why it is not working.
I tried adding __init__.py to the folder but didn’t help.