diff --git a/src/arff/arff.cpp b/src/arff/arff.cpp
index b6cd3ea..be00b51 100644
--- a/src/arff/arff.cpp
+++ b/src/arff/arff.cpp
@@ -114,10 +114,10 @@ namespace ARFF {
         parser >> token;
         // Signifies beginning of data
         // Might add a boolean later to mark this
-        if (token == "@data") {
+        if (token == "@data" || token == "@DATA") {
             return;
         }
-        if (token == "@relation") {
+        if (token == "@relation" || token == "@RELATION") {
             parser >> token;
             relation = token;
             LogInfo("ARFF/Attribute", "Relation set: " + relation);