Fix bug as pointed out by @lividtech