Fix bug: #370