made code change to match apiary