fix unit test by removing contains