Temporary fix for ok_test