fixed a small bug hopefully