diff --git a/app/rumblegraphviewer.h b/app/rumblegraphviewer.h
index d5a624b..c4b071c 100644
--- a/app/rumblegraphviewer.h
+++ b/app/rumblegraphviewer.h
@@ -1,29 +1,30 @@
-#ifndef RUMBLEGRAPHVIEWER_H
-#define RUMBLEGRAPHVIEWER_H
-
-#include "zonefile.h"
-#include <QWidget>
-
-namespace Ui {
-class RumbleGraphViewer;
-}
-
-class RumbleGraphViewer : public QWidget
-{
-    Q_OBJECT
-
-public:
-    explicit RumbleGraphViewer(QWidget *parent = nullptr);
-    ~RumbleGraphViewer();
-
-    void SetEntryCount(quint32 aCount);
-    void SetRumbleGraphFile(const std::shared_ptr<RawFile> aRawFile);
-    void SetZoneFile(std::shared_ptr<ZoneFile> aZoneFile);
-
-private:
-    Ui::RumbleGraphViewer *ui;
-    quint32 mEntryCount;
-    std::shared_ptr<RawFile> mRumbleGraphFile;
-};
-
-#endif // RUMBLEGRAPHVIEWER_H
+#ifndef RUMBLEGRAPHVIEWER_H
+#define RUMBLEGRAPHVIEWER_H
+
+#include "xrawfile.h"
+#include "zonefile.h"
+#include <QWidget>
+
+namespace Ui {
+class RumbleGraphViewer;
+}
+
+class RumbleGraphViewer : public QWidget
+{
+    Q_OBJECT
+
+public:
+    explicit RumbleGraphViewer(QWidget *parent = nullptr);
+    ~RumbleGraphViewer();
+
+    void SetEntryCount(quint32 aCount);
+    void SetRumbleGraphFile(const XRawFile *aRawFile);
+    void SetZoneFile(ZoneFile* aZoneFile);
+
+private:
+    Ui::RumbleGraphViewer *ui;
+    quint32 mEntryCount;
+    const XRawFile* mRumbleGraphFile;
+};
+
+#endif // RUMBLEGRAPHVIEWER_H