Fixed missing os import in setup.