Fix unintended bug