Add missing EOF check