minor code fix.