Fix story bucket unit tests