Fix test (poorly)