fix the test :bear: