[BUGFIX] Fix test