Attempt at fixing test