fixed the fucking bug