ah, fixed the bug