neglected an update step. should fix #3