fixing errors in tests