replace `Stage.where(:id => 1)` with `Stage.first`