modif pour issue#2

fixed #2