Update tests with new namespace