fix problem: #4