Fixing timing for finite difference by adding a wait