fix issue with stage version