Add missing check