fixed pickle error