updated the test by correcting the key to not to look for total