Class syntax fix