fix for new spec