Fix mistake made in 3 unit tests

They were testing the wrong methods.