fix bug....