added changes to test