fix take
fix take and add test