Minor refactor

Noticed this while taking a cursory glance through the code.