Remove unnecessary import
Fix reference