temporary fix to unpickling error