[PATCH] Fix WARN_ON / WARN_ON_ONCE regression