Updated key management test running