Fix mess made in previous commit