Update to fix refresh issue