Update. Fixes #3