improve test rake task