Fixed small bug
update