fix for tests - now passing locally.