Fix type that breaks build :(