better logic for no input stage