Forgot to import class