Fix EOF bug