Revert "Fixing test"

This reverts commit d56f91dbc798be6fd4b80bd6078c29852fa03474.