fix small bug related to refactoring.
fix small bug related to refactoring.

twitter/ambrose,tim777z/ambrose,twitter/ambrose,tim777z/ambrose,tim777z/ambrose,twitter/ambrose,tim777z/ambrose,twitter/ambrose,tim777z/ambrose,twitter/ambrose