Fix un/pickling extenions