Whoops, forgot to add an if check.