Fixing missing use statement