Fixed import problem with pickle

A module was moved and this prevented some data being loaded. This was
fixed