Update test to check for warning messages