Another import fix

Last PR of Q3. I promise!