Fixing bad use statement after server refactoring