get tests to pass after okm changes