diff --git a/src/Main.java b/src/Main.java
index ad07f78..66ebd93 100644
--- a/src/Main.java
+++ b/src/Main.java
@@ -26,8 +26,8 @@ public static void main(String[] args) throws ExecutionException, InterruptedExc
 //        }
 
         ExecutorService executorService = Executors.newFixedThreadPool(2);
-        RandomNumberGenerator random1 = new RandomNumberGenerator();
-        RandomNumberGenerator random2 = new RandomNumberGenerator();
+        RandomNumberGenerator random1 = new RandomNumberGenerator<Integer>();
+        RandomNumberGenerator random2 = new RandomNumberGenerator<Integer>();
 
         Future<Integer> randomNum1 = executorService.submit(random1);
         Future<Integer> randomNum2 = executorService.submit(random2);
@@ -38,4 +38,4 @@ public static void main(String[] args) throws ExecutionException, InterruptedExc
         System.out.println("Sum of random numbers : " + (num1 + num2) +
                 " THREAD: " + Thread.currentThread().getName());
     }
-}
\ No newline at end of file
+}
diff --git a/src/RandomNumberGenerator.java b/src/RandomNumberGenerator.java
index 11c481a..847cdac 100644
--- a/src/RandomNumberGenerator.java
+++ b/src/RandomNumberGenerator.java
@@ -1,7 +1,7 @@
 import java.util.Random;
 import java.util.concurrent.Callable;
 
-public class RandomNumberGenerator implements Callable {
+public class RandomNumberGenerator implements Callable<Integer> {
     @Override
     public Integer call() throws Exception {
         Random random = new Random();