Updated tests to use more accurate assertions