Minor bug in previous commit