Fix IOError bug