[BUGFIX] Fix copied code