fixed bug, sorry