add switch for test