fix bugs with test.
add limitation