Updating Junits to reflect source changes