Fixed bug from purge class removal