Fix #18 | Thanks to @azag0