fixing success message on test