One minor bug fix