fixed the bug hopefully