Type signature fix