Update tests to use proper namespace