(fix) small bug agin :/