Review comments + test
PTX-level test