Fix #377, fix #378