Fix Issue 1

fix issue #1