Bugfix I forgot to push earlier
Missing include fix hadn't been pushed either