Test class changed due to warning