Fixed errors in test class