I think this if should be part of the `__init__` function... if it's where it's now probably it won't be from __init__ anymore... no?