fixing function signature