Update to work with Wedge Alpha 3