Fixed bug resulting from moving method from other class.