fix: remove __main__ (used for tests)