Updated tests file. #6