Update tests for ok.py
Change paths to attributes in ok.py tests