Fixed regressions on the `init_socks.py` demo.
Fixed some broken and unnecessary imports.