Fix type thanks to @lkuper