Fixed the bug in the previous commit

Signed-off-by: Shahbaz Youssefi <ShabbyX@gmail.com>