Add back missing check