updated test so it seems to run fine