Fix import in tests.

Fix const names.