Fixed bug with missing include file