Fixed improper import statement