Fixes error in success test