little code
fix #3