diff --git a/setup.py b/setup.py
index f19447a..6af0f3d 100644
--- a/setup.py
+++ b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name='Skperopt',
-    version='0.0.695',
+    version='0.0.7',
     packages=["skperopt"],
     url='https://github.com/lewis-morris/Skperopt',
     license='MIT',
@@ -15,7 +15,7 @@
     author_email='lewis.morris@gmail.com',
     description='Hyperopt Wrapper',
     install_requires=requirements,
-    download_url = 'https://github.com/lewis-morris/Skperopt/archive/0.0.695.tar.gz',
+    download_url = 'https://github.com/lewis-morris/Skperopt/archive/0.0.7.tar.gz',
     keywords=['hyperopt-wrapper', 'hyperparameter'],
     classifiers=[
         'Development Status :: 5 - Production/Stable',
diff --git a/skperopt/search.py b/skperopt/search.py
index bfbf8ec..f014435 100644
--- a/skperopt/search.py
+++ b/skperopt/search.py
@@ -113,7 +113,7 @@ def scorer_is_better(test_type, new_score, old_score):
             return True
         else:
             return False
-    elif test_type == "rmse":
+    else:
         if old_score >= new_score:
             return True
         else:
@@ -139,6 +139,8 @@ def get_score(y_true, y_pred, scorer):
             score_list.append(f1_score(y_true, y_pred, average='macro'))
         elif score_type == "rmse":
             score_list.append(mean_squared_error(y_true,y_pred,squared = False))
+        elif score_type == "mse":
+            score_list.append(mean_squared_error(y_true, y_pred, squared=True))
         elif score_type == "auc":
             score_list.append(roc_auc_score(y_true, y_pred, average="macro"))
         elif score_type == "accuracy":