Unit tests being updated