Minor Code Tweak