Fixed refresh issue. Minor improvement.