Fix fucking bug