Update test case to fix improper messaging