Suggestions for fixing test?