fix for failing tests