Improve tests to check function signature too