Fix imports of test script