Fixes unit test errors by changing hardcoded values to the values the new code outputs. The new code is believed to be correct so this should be alright.