fixes error in tests