Updated unity test