Fix this class who was broken by the previous commit