Fix issue #88.

Fix issue #88. When argument is an array, we should `flatten` it first.