fix teh bug