Fixed build error due to missing include