Put back class removed by mistake