fixed bug with wrong import statement