Small code fix.