fix for broken test