Fixed one more test-then-assert for new af-api