Fix of test importing modules