Add test to cover less import statements

This commit adds coverage for the new functionality introduced in PR #61