Fix imports

Fix for 0.5 (used by DDSC)