Fix tests

Dependency of https://github.com/transmart/transmartApp/pull/82