Bugfix: missing #include