Fix compiler warning.
Fix import causing failed build.