Fixed error arisen during PR