small bug. atempt to fix