Test import fix