Added missing check