[dev]adjust code for test