Adjust specs to use promise