Fixed a mistake in the broken code.

I was checking in the wrong place.