Fix failing AP test