Fixing test imports.