Fixes missing use statement