Fix relative import issue