really fixed the bug