diff --git a/DESCRIPTION b/DESCRIPTION
index 8fb8004..525efb7 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -1,6 +1,6 @@
 Package: tMAE
 Title: Tests and visualizes for Mono-allelic expressed variants
-Version: 1.0.0
+Version: 1.0.1
 Authors@R: c(
     person(given = "Vicente",
            family = "Yepez",
diff --git a/R/add_gnomAD_AF.R b/R/add_gnomAD_AF.R
index 303750b..4e48ef3 100644
--- a/R/add_gnomAD_AF.R
+++ b/R/add_gnomAD_AF.R
@@ -56,10 +56,7 @@ add_gnomAD_AF <- function(data,
   
   # Compute the MAX_AF (why do we change col names?)
   if(any(c("AF", "AF_popmax") %in% colnames(res))){
-    if("AF_popmax" %in% colnames(res)){
-      res[,MAX_AF:=AF_popmax]
-    } else {
-      res[,MAX_AF:=AF]
+    res$MAX_AF <- apply(res[, ..populations], 1, max, na.rm = T)
     }
     
     # Replace Inf with NA