Attention correction

Fixing attention. Added a basic test value.
Perhaps we should use the `blocks.bricks.DEFAULT_SEED` here. But I don't know for sure, so let's merge.