@linki You might be able to DRY this up slightly by moving this to the `before` block.