Fix incorrect import. Closes salilab/imp#861.