1 files changed, 3 insertions, 2 deletions
diff --git a/archivebrowsermodel.cpp b/archivebrowsermodel.cpp
index 08a0752..5da95f1 100644
--- a/archivebrowsermodel.cpp
+++ b/archivebrowsermodel.cpp
@@ -11,6 +11,7 @@
 
 #include "archivebrowsermodel.h"
 #include "smtreeitem.h"
+#include "smglobals.h"
 #include "helper.h"
 
 ArchiveBrowserModel::ArchiveBrowserModel(const QStringList &headers, QObject *parent) : SmTreeModel(headers, parent), mNumFields(8) {
@@ -52,10 +53,10 @@ QVariant ArchiveBrowserModel::data(const QModelIndex &index, int role) const {
         if(col == TotalSize){
             qint64 s = item->data(TotalSize).toDouble();
             QColor retval = Qt::green;
-            if(s > Q_INT64_C(1024 * 1024 * 1024 * 2)){
+            if(s > DVDSIZE){
                 retval = Qt::cyan;
             }
-            if(s > Q_INT64_C(1024 * 1024 * 1024 * 4)){
+            if(s > DVDSIZE){
                 retval = Qt::red;
             }
             return retval;