Fixes missing import class