Minor source code improvement