Expand - Minor code fix