fixed broken test