1 files changed, 9 insertions, 9 deletions
diff --git a/test/openpower-pels/generic_section_test.cpp b/test/openpower-pels/generic_section_test.cpp
index 9fb6d9a..7cb9949 100644
--- a/test/openpower-pels/generic_section_test.cpp
+++ b/test/openpower-pels/generic_section_test.cpp
@@ -8,13 +8,13 @@ using namespace openpower::pels;
 TEST(GenericSectionTest, UnflattenFlattenTest)
 {
     // Use the private header data
-    auto data = pelDataFactory(TestPelType::privateHeaderSimple);
+    auto data = pelDataFactory(TestPELType::privateHeaderSection);
 
-    Stream stream(*data);
+    Stream stream(data);
     Generic section(stream);
 
     EXPECT_EQ(section.header().id, 0x5048);
-    EXPECT_EQ(section.header().size, data->size());
+    EXPECT_EQ(section.header().size, data.size());
     EXPECT_EQ(section.header().version, 0x01);
     EXPECT_EQ(section.header().subType, 0x02);
     EXPECT_EQ(section.header().componentID, 0x0304);
@@ -22,11 +22,11 @@ TEST(GenericSectionTest, UnflattenFlattenTest)
     const auto& sectionData = section.data();
 
     // The data itself starts after the header
-    EXPECT_EQ(sectionData.size(), data->size() - 8);
+    EXPECT_EQ(sectionData.size(), data.size() - 8);
 
     for (size_t i = 0; i < sectionData.size(); i++)
     {
-        EXPECT_EQ(sectionData[i], (*data)[i + 8]);
+        EXPECT_EQ(sectionData[i], (data)[i + 8]);
     }
 
     // Now flatten
@@ -34,16 +34,16 @@ TEST(GenericSectionTest, UnflattenFlattenTest)
     Stream newStream(newData);
     section.flatten(newStream);
 
-    EXPECT_EQ(*data, newData);
+    EXPECT_EQ(data, newData);
 }
 
 TEST(GenericSectionTest, BadDataTest)
 {
     // Use the private header data to start with
-    auto data = pelDataFactory(TestPelType::privateHeaderSimple);
-    data->resize(4);
+    auto data = pelDataFactory(TestPELType::privateHeaderSection);
+    data.resize(4);
 
-    Stream stream(*data);
+    Stream stream(data);
     Generic section(stream);
     ASSERT_FALSE(section.valid());
 }