Fixed a bug, included missing library