just write 
return stage != null ? stage.getStageId() : null;
spare some lines :  )