fix bug introduced by beautify