Fix bug: cải tiến để phù hợp với 2.1.3