Update test case for ZK-2458