Fix missed relative import