Fixed accidentally broken test