Fix for controllers in production