bugfix: add a missing import