Fixes missing import statement