Fixing unit test after about you addition