Add task to run test
Fix broken test