Clean up test to remove unnecessary variable