Add guard for running tests