Staging bug fix