Fixed incorrect imports

Can't compile project without these changes