Small Fixed to previous patch