diff --git a/core-java-modules/core-java-concurrency-advanced-3/src/main/java/com/baeldung/atomicstampedreference/StampedAccount.java b/core-java-modules/core-java-concurrency-advanced-3/src/main/java/com/baeldung/atomicstampedreference/StampedAccount.java
index 1a46e1ba52..415b24738a 100644
--- a/core-java-modules/core-java-concurrency-advanced-3/src/main/java/com/baeldung/atomicstampedreference/StampedAccount.java
+++ b/core-java-modules/core-java-concurrency-advanced-3/src/main/java/com/baeldung/atomicstampedreference/StampedAccount.java
@@ -9,13 +9,11 @@ public class StampedAccount {
     private AtomicStampedReference<Integer> account = new AtomicStampedReference<>(0, 0);
 
     public int getBalance() {
-        return this.account.get(new int[1]);
+        return account.getReference();
     }
 
     public int getStamp() {
-        int[] stamps = new int[1];
-        this.account.get(stamps);
-        return stamps[0];
+        return account.getStamp();
     }
 
     public boolean deposit(int funds) {