Improved unit test 3