[FIX] import which breaks tests