Fix import statement (forgot to change after moving class)