From 761926db2ce0c619c9fcd3a9a4430a92eb5db905 Mon Sep 17 00:00:00 2001
From: Yiqun Lin <yqlin@apache.org>
Date: Thu, 9 Nov 2017 18:59:10 +0800
Subject: [PATCH] HDFS-12793. Ozone : TestSCMCli is failing consistently.
 Contributed by Chen Liang.

---
 .../org/apache/hadoop/ozone/scm/container/ContainerMapping.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/ozone/scm/container/ContainerMapping.java b/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/ozone/scm/container/ContainerMapping.java
index b77c5b1b28..98d6396a2a 100644
--- a/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/ozone/scm/container/ContainerMapping.java
+++ b/hadoop-hdfs-project/hadoop-hdfs/src/main/java/org/apache/hadoop/ozone/scm/container/ContainerMapping.java
@@ -311,7 +311,7 @@ public class ContainerMapping implements Mapping {
       ContainerInfo updatedContainer = containerStateManager
           .updateContainerState(containerInfo, event);
       containerStore.put(dbKey, updatedContainer.getProtobuf().toByteArray());
-      return containerInfo.getState();
+      return updatedContainer.getState();
     } catch (LeaseException e) {
       throw new IOException("Lease Exception.", e);
     } finally {