Update to try to fix end of input error