fixed a few things on tests