Add a missing import statement