fix bug - Fixed #4