Add class back in to fix test failure

Although, I question where this links to in admin now. Should we point it at How gov works?