Fixing testing logic.