fix extension test imports