Cleanup: fix silly bug