try to fix litle bug