Fix for page reload