Fix missing include in controllers