fixed broken test due to changes