Fixing ex1 tests