diff --git a/torch_geometric/datasets/brca_tgca.py b/torch_geometric/datasets/brca_tgca.py
index e699c141316a..e5778bc2ce25 100644
--- a/torch_geometric/datasets/brca_tgca.py
+++ b/torch_geometric/datasets/brca_tgca.py
@@ -92,7 +92,7 @@ def process(self):
 
         data_list = []
         for x, y in zip(graph_feat, graph_label):
-            data = Data(x=x, edge_index=edge_index, y=y)
+            data = Data(x=x.view(-1, 1), edge_index=edge_index, y=y)
 
             if self.pre_filter is not None and not self.pre_filter(data):
                 continue