Should this be `dtype=self.dtype`?