Fixes missing import.