Minor code fix.