Fixed input tests.

Fixes #8