fixing test to more accurately test