Fixing minor issue with `take`.