fixes an import error after moving the datastructures