fix: input array, add: test