fix bug ..