Fixing test

Related to #2355