Fix for test imports.