Catch correct warning

Why these changes?