Fixed a bug in the implementation.