diff --git a/project/src/helpers/WeightedRandomHelper.ts b/project/src/helpers/WeightedRandomHelper.ts
index da9dcf3e..4c3ba03c 100644
--- a/project/src/helpers/WeightedRandomHelper.ts
+++ b/project/src/helpers/WeightedRandomHelper.ts
@@ -42,18 +42,23 @@ export class WeightedRandomHelper
      * @param {number[]} weights
      * @returns {{item: any, index: number}}
      */
-    public weightedRandom(items: string | any[], weights: string | any[]): { item: any; index: number; }
+    public weightedRandom(items: any[], weights: any[]): { item: any; index: number; }
     {
+        if (!items || items.length === 0)
+        {
+            throw new Error("Items must not be empty");
+        }
+
+        if (!weights || weights.length === 0)
+        {
+            throw new Error("Item weights must not be empty");
+        }
+
         if (items.length !== weights.length)
         {
             throw new Error("Items and weight inputs must be of the same length");
         }
 
-        if (!items.length)
-        {
-            throw new Error("Items must not be empty");
-        }
-
         // Preparing the cumulative weights array.
         // For example:
         // - weights = [1, 4, 3]