Fixed a bug I introduced with previous change.