Update/tests

@ianstormtaylor