diff --git a/OOP/Java/Lab/Week4/TraceNorm.java b/OOP/Java/Lab/Week4/TraceNorm.java
index a403045..15b5ac5 100644
--- a/OOP/Java/Lab/Week4/TraceNorm.java
+++ b/OOP/Java/Lab/Week4/TraceNorm.java
@@ -13,7 +13,10 @@ class TraceNorm {
         System.out.println("Enter the elements of the matrix:");
         for (int i = 0; i < n; i++) {
             for (int j = 0; j < n; j++) {
-                matrix[i][j] = sc.nextDouble();
+                System.out.print(
+                    "Enter the element at (" + i + "," + j + "): "
+                );
+                matrix[i][j] = sc.nextInt();
             }
         }