Added import...

...to fix compilation error