updated test to remove assert