fix loading visualization