fixing update class