better check, added one missed class