Fix build with correct import.