Update tests to work with Guardian deps