Fix 'needInput' functionality