fix bug: #2