a small code cleanup #fix