Fixed a bug when refactoring wavefunc slicing into helper function

[1tnguyen]

The single amplitude was returned in an one-element vector. I accidentally use an invalid index to get the imaginary part.

Signed-off-by: Thien Nguyen nguyentm@ornl.gov