Just forgot the class