Fix missing import
Improve function name