small update
modify test code