Moved include to where it's needed