- updating test to make it work for tasks