[PATCH 2/4] Add missing check