[BUGFIX] Fixed tests.