Fixed wrong function signatures when calling the transform layer