Fix bug. Thanks user chronossc at github