[BUGFIX] Fix last failing test