fix issue 5

fix issue 5