Dirty fix for test