Fix display of loading element