Syntax fix on class