Updated tests for XUnit