Add production fixes