Fixing import ref