add back that check