Minor fix
minor fix.
Fixed #301.
Fixed #302.