fixed bug i think