minor code fix