add forgotten class during previous commit