fixes tests

@adamggg