Added a missing library import