Bug fix for removed file include