Fixing decorator issue in _init.py