Fixed a bug in test class.