Bugfix for input stage