Fix running worker as a module