Fix token refresh