Replace AtomicReference usage with AtomicReferenceFieldUpdater