Update tests to use namespace