minor code polish