Fix stage checking

[a-ws-m]

Quick PR that solves #272 . I forgot I'd done this a while ago!

[orbeckst]

closing and opening to cycle CI

[orbeckst]

cyle CI (after re-enabling)

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 82.39%. Comparing base (c719914) to head (3c0b00c). Report is 8 commits behind head on develop.

Additional details and impacted files

@@             Coverage Diff             @@
##           develop     #278      +/-   ##
===========================================
+ Coverage    81.65%   82.39%   +0.74%     
===========================================
  Files           15       15              
  Lines         2017     2017              
  Branches       310      310              
===========================================
+ Hits          1647     1662      +15     
+ Misses         275      262      -13     
+ Partials        95       93       -2     

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.