fix tests. optiml tests passing