Fix a problem that it doesn't work when installed by pip