fix bug;