Code syntax fix