updated test to assert