adding changes to tests