Fixiing broken test