Apply correction on mock.dart