Fix some issues

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