Fix error on reload