diff --git a/sstmap/_sstmap_probable.cpp b/sstmap/_sstmap_probable.cpp
index 17136aa..3e7b0ac 100644
--- a/sstmap/_sstmap_probable.cpp
+++ b/sstmap/_sstmap_probable.cpp
@@ -21,7 +21,7 @@ using namespace std;
     }
 #endif
 
-int prob(string infile, string outfile) {
+void prob(string infile, string outfile) {
     /*
     if (argc <= 1) {
         cerr << "\nUSAGE:\n\n"
diff --git a/sstmap/kdhsa102.cpp b/sstmap/kdhsa102.cpp
index 925a3e2..e917e29 100644
--- a/sstmap/kdhsa102.cpp
+++ b/sstmap/kdhsa102.cpp
@@ -572,10 +572,10 @@ double kdtree::run_tree_trans(std::vector<double > &cls) {
     double R = 8.314472;
     double pi = 3.14159265359;
 
-    int fcount = 10000;
+    //int fcount = 10000;
 
     for (int i = 0; i < numvals; i++) {
-        gd += log((0.0329223149*fcount*4*pi*pow(dh[i], 3))/3);
+        gd += log((0.0329223149*numvals*4*pi*pow(dh[i], 3))/3);
     }
 
     s = R*T*0.239*(gd/numvals + 0.5772156649)/1000;