update for making the test work(sort of)