Fixed issued with local storage