fix omitted import statement