PhotonVERSE: Minor bug fixes

Fixed a minor bug in stage 5-4