Fix relative import that was causing travis CI build failure