Based on dense_design_matrix, this should be self.X, not self.X_space