Any better way to do this since the line changed for that test?