TST: fixed new test and added baseline PNG