minor code clean up, h/t to @EvanHahn