Missed a change, forgot to import custom wrapper