modify a little for test.