- small code cleanup based on review suggestion