Fix import and usage.