fixed last minute bug where an import was forgotten for a certain code path