Fixed the test to match current way of setting input.