add guard for test