fixed bug hopefully