Add import to fix build on some machines