Fix small error (in previous commit)