Fixing the tests to actually run the code