Fix tests for testling