Minor code change for Paul's Review