Restore missed conditional