It is common to put additional logic in afterEach step (to return to primary state, make additional repetitive checks, etc).
With current implementation, when the test fails, afterEach is being executed and only THEN screenshot is taken. It means that other screen could be opened already, so screen where bug happened is not visible anymore. This way it is harder to analyze bug failures, as you do not have corresponding screenshot.

Please implement taking screenshot at the start of afterEach statement.