Adds missing import statement