Little fix for the last commit

Signed-off-by: iceandfire <arhamjamal@gmail.com>