fixing that bugggg