Fixed: make it importable