fix bug with