Change `use`-statement to fix compilation