[CHORE] fixes tests for 0.2.0

Signed-off-by: Brian Nelson <yonbeastie@gmail.com>