Fixed final compilation error (wrong import)