Should #ifdef(TEST) around this