Fixed import, removed logging