Add test for the two methods that were untested