fixed the bug (hopefully)