Fix test running against local version