Fixed staging bug