fixed the bug possibly