fix the use_ok to test the correct module