some fix
add test code