[BUGFIX] Use correct backend layout in FE