Fixing breaking test