update test case for klass