Make test not crash //TODO fix me properly