Adds missing import