Fix Class capitalisation