Fixed a bug with restricted_to not being found