Fix test to use puback