Fixing a bug I added in previous commit