making tests fit the 0.4.0 change