Update unit test for updated error message