Forgot some class redecaration