Update tests
Remove public