Setup tests to use guard