Fixed a test and was unable to get another function to give correct output