Fix setup for testling