fix: support afterEach

Closes #2