[TASK] Fix rake task for running tests