Fixed compilation that was missing import statement