diff --git a/syngen_floorplan_multipath.py b/syngen_floorplan_multipath.py
index 212811f..34188a9 100644
--- a/syngen_floorplan_multipath.py
+++ b/syngen_floorplan_multipath.py
@@ -114,7 +114,8 @@ def generateData(floormap, tx_locs, args):
             for j in range(0, rx_locs.shape[0], args.procnum):
                 procs = []
                 results = Manager().dict()
-                for kk in range(args.procnum):
+                actual_num = min(args.procnum, rx_locs.shape[0] - j)
+                for kk in range(actual_num):
                     rx_loc = SpaceBlock(rx_locs[j + kk, 0], rx_locs[j + kk, 1])
                     proc = Process(
                         target=log_gamma_floorplan_multi,
@@ -124,7 +125,7 @@ def generateData(floormap, tx_locs, args):
                     procs.append(proc)
                 for proc in procs:
                     proc.join()
-                rss_vec.extend([results[kk] for kk in range(args.procnum)])
+                rss_vec.extend([results[kk] for kk in range(actual_num)])
 
                 # rx_loc = SpaceBlock(rx_locs[j, 0], rx_locs[j, 1])
                 # rss, multipaths = log_gamma_floorplan(