Fix failing test case when only package is imported