Fixed a small bug.

Signed-off-by: Amrykid <amrykid@gmail.com>