Fix syntax a bit