I think I fixed the bug