Fixed bug.

Signed-off-by: Afsheen Ghorashy <arghorashy@gmail.com>