Add conditional for test.c