Fix bug in previous commit - oops.