fixed error with self.class