Function signature fix