Add missing argument to self.sample call