fix for failing test