Fixed using array_map