Use self.class

Almost entirely a matter of personal preference here I admit