Fix test
Added assertion