minor changes attempting to fix broken test