actually fixed the bug