Fix: missing use statement