Goddammit. Forgot a class.