Refactor ShellMock to only apply during spec examples