Fix issue  #2

Fix #2