Minor Code Change