fix bug for Ubuntu