updated edit method for staging display issue