fixed bug yo