Fixed error on test class.