updated test back to way it was