Add Guard for tests