Putting back the include which was removed for testing