Update three sigma test to use new API