fix #6 issue for milestone 0.0.3