Fixed tests to run again