add check for original class