main.py

Didn't notice you're actually passing back the created object :)