One minor code modification