Updating unit testing stuff