Fixed localvirial bug in localpressure branch, caused by incorrectly placed graph shift calls

Change-Id: I085062d3feffd9ef7f22e0ae9b7f7aff95c6c53e