Fix small bug in offset / take