Add extra tests for StagesController; fix model update on frontend