Dirty hack to work around circular import.