Simplify importing extensions in unit tests