Oops
Removed a needed import in the previous commit

Signed-off-by: PipelinerArd <kristoferard@gmail.com>