Adds missing null check, whoops