fix tests for checkpoint api