Update import statement so that project compiles