Fix to remove import of tests in __init__