Small bug fix

Signed-off-by: Ramnique Singh <mail@ramniquesingh.name>