Fixed bug in test class.