Import the right package.

This fixes tests.