minor code fix

see the file change (its really minor!)