forgot to add the test changes last time... sorry