Fixed the code after EOF token error