Fixed pickle error