Fixing issue with not running some tests.